from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms

import cv2

# python的用法--> tensor数据类型
# 通过 transforms.ToTensor去解决俩个问题
# 1.transform该如何使用(python)
# 2.为什么需要Tensor数据类型

# 绝对路径-->E:\PycharmProjects\learn_torch\dataset\train\ants_image\0013035.jpg
# 相对路径-->dataset/train/ants_image/0013035.jpg
# img_path = "dataset/train/ants_image/0013035.jpg"  # 相对路径使用报错，绝对路径不会-->22-04-17
img_path_abs = r"E:\PycharmProjects\learn_torch\dataset\train\ants_image\0013035.jpg"  # 前面加一个r取消转义-->即可以不用将\变成\\
img = Image.open(img_path_abs)
# print(img)  # 查看文件类型

writer = SummaryWriter("logs")

# 1.transform该如何使用(python)-->解决如下
tensor_trans = transforms.ToTensor()  # 只要引用-->ToTensor就会返回class ToTensor的一个对象
tensor_img = tensor_trans(img)  # 当光标在（）时，按下ctrl+p就可以知道当前需要什么样的参数

# print(tensor_img)

writer.add_image("Tensor_img", tensor_img)

writer.close()
